package com.leetcode.easy;

import com.leetcode.entity.TreeNode;

import java.util.*;

/**
 * @Classname SolutionLevelOrderBottom
 * @Description 二叉树的程序遍历2 从下往上输出
 * @Date 2020/9/3 15:21
 * @Created by zfl
 */
public class SolutionLevelOrderBottom {

    /**
     * @param root
     * @return
     * 执行用时：1 ms, 在所有 Java 提交中击败了98.21%的用户
     * 内存消耗：39.8 MB, 在所有 Java 提交中击败了88.27%的用户
     */
    public List<List<Integer>> levelOrderBottom(TreeNode root) {
        if(root == null) {
            return new ArrayList<>();
        }
        Stack<List<Integer>> stack = new Stack<>();
        List<List<Integer>> result = new ArrayList<>();
        Queue<TreeNode> q = new LinkedList<>();
        int count;
        q.add(root);
        while(!q.isEmpty()) {
            count = q.size();
            List<Integer> list = new ArrayList<>();
            while (count > 0) {
                TreeNode ret = q.remove();
                list.add(ret.val);
                if(ret.left != null) {
                    q.add(ret.left);
                }
                if(ret.right != null) {
                    q.add(ret.right);
                }
                count--;
            }
            stack.add(list);
        }
        while (!stack.isEmpty()) {
            result.add(stack.pop());
        }
        return result;
    }

    /**
     * @param root
     * @return
     * 使用有序链表，添加结果集时往头部插入即可
     * 执行用时：1 ms, 在所有 Java 提交中击败了98.21%的用户
     * 内存消耗：39.8 MB, 在所有 Java 提交中击败了82.41%的用户
     */
    public List<List<Integer>> levelOrderBottom1(TreeNode root) {
        if(root == null) {
            return new ArrayList<>();
        }
        List<List<Integer>> ans = new LinkedList<>();
        Queue<TreeNode> queue = new LinkedList<>();
        queue.offer(root);
        while (!queue.isEmpty()) {
            int size = queue.size();
            List<Integer> list = new ArrayList<>();
            while (size > 0) {
                TreeNode node = queue.poll();
                list.add(node.val);
                if(node.left != null) {
                    queue.offer(node.left);
                }
                if(node.right != null) {
                    queue.offer(node.right);
                }
                size--;
            }
            ans.add(0,list);
        }
        return ans;
    }
}
